Search CTRL + K

Repository

Git

如何利用yaml 构建一个合适的科研工程项目的config?

import argparse
import yaml



def main()
  pass




if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Run the program')
    parser.add_argument("--config",type=str,default="./config/config.yaml")
    # TODO add more arguments maybe?
    args = parser.parse_args()
    with open(args.config,"r") as f:
        config = yaml.safe_load(f)
    main(config,args)


注意 config.yaml 文件传给main 的 config 就是一个字典,这样实际上我们是可以通过一些实际的类似如下的方式传递参数


def function(arg1,arg2,arg3,arg4,**kwargs):
  pass

# config 是一个字典,key 里面包含有 arg1,arg2,arg3,arg4 来传递参数,多出来的key 会传递给 kwargs 通过这样的方式实现参数传递
function(**config)